from ship_envs.environment import imazu_problem
from ship_envs.environment.ImazuProblemEnv import ImazuProblemEnv


def get_case_by_index(case_index):
    case_name = 'case' + str(case_index)
    return getattr(imazu_problem, case_name)


def new_case_env_by_index(case_index):
    case = get_case_by_index(case_index)
    return ImazuProblemEnv(case())
